ARG BASE_IMAGE=520713654638.dkr.ecr.us-west-2.amazonaws.com/sagemaker-rl-tensorflow:ray0.6.5-gpu-py3
FROM $BASE_IMAGE
RUN pip install --upgrade pip && pip install --no-cache-dir xpress networkx